package demo.standard;

import java.util.ArrayList;
import java.util.List;

public class Caretaker {
    private List<Memento> mementos = new ArrayList<>();

    public void add(Memento memento) {
        mementos.add(memento);
    }

    public Memento undo() {
        if (mementos.isEmpty()) {
            return null;
        }
        int lastIndex = mementos.size() - 1;
        Memento memento = mementos.get(lastIndex);
        mementos.remove(lastIndex);
        return memento;
    }
}
